from matplotlib import pyplot as plt

plt.figure()
plt.plot(train_recon_loss, 'b', label='Recon_loss')
plt.ylabel('Recon_loss')
plt.xlabel('iter_num')
plt.legend()
plt.savefig(os.path.join(imgPath, "1_recon_loss.jpg"))